package com.cskaoyan.mr.relation;

import org.apache.hadoop.io.IntWritable;
import org.apache.hadoop.io.Text;
import org.apache.hadoop.mapreduce.Reducer;

import java.io.IOException;
import java.util.Iterator;

/**
 * @ClassName RelationReducer
 * @Description: TODO
 * @Author 远志 zhangsong@cskaoyan.onaliyun.com
 * @Date 2023/2/13 15:58
 * @Version V1.0
 **/
public class RelationReducer extends Reducer<Text, IntWritable, Text, IntWritable>
{
    /**
     *
     * @param key 组合之后形成的名称，比如郭靖-黄蓉
     * @param values 之间关系的描述；直接好友用0表示；潜在间接好友用1表示
     * @param context
     * 处理的过程就是将values的长度和数字相加进行比较；如果大小不一致，说明中间出现了0
     * 表示二人是直接好友，不需要推荐
     * @throws IOException
     * @throws InterruptedException
     */
    @Override
    protected void reduce(Text key, Iterable<IntWritable> values, Context context) throws IOException, InterruptedException {
        Iterator<IntWritable> iterator = values.iterator();
        int length = 0;
        int sum = 0;
        while (iterator.hasNext()){
            length ++;
            IntWritable writable = iterator.next();
            sum += writable.get();
        }
        if(length == sum){
            //二人是间接好友关系，可以推荐 key是二人的关系 value是二人之间的共同好友数
            context.write(key, new IntWritable(length));
        }
    }
}
